/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */

package br.com.krossk.servlets;

import br.com.krossk.dao.RepositorioDAO;
import br.com.krossk.jpa.JPAFactory;
import br.com.krossk.model.Arquivo;
import br.com.krossk.model.Repositorio;
import br.com.krossk.model.Usuario;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.OutputStream;
import javax.persistence.EntityManager;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

/**
 *
 * @author Henrique
 */
public class RecoverFileServlet extends HttpServlet{
   
    /** 
     * Processes requests for both HTTP <code>GET</code> and <code>POST</code> methods.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    protected void processRequest(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException {
        
        int idArquivo = Integer.parseInt(request.getParameter("idArquivo"));
        HttpSession session = request.getSession();
        Usuario u = null;
       
        u = (Usuario)session.getAttribute("usuario");

        EntityManager em = JPAFactory.getEntityManager();
        Arquivo arquivo = em.find(Arquivo.class, idArquivo);
        if(!arquivo.getUsuarioCollection().contains(u)){
            arquivo.getUsuarioCollection().add(u);
            em.getTransaction().begin();
            em.merge(arquivo);
            em.getTransaction().commit();
        }

        em.close();
       
        response.setContentType(arquivo.getTipoArquivo());//Pesquise os mime-types para saber o tipo correto
        Repositorio rep = RepositorioDAO.getInstance().getById(arquivo.getIdArquivo());
        ByteArrayInputStream in = new ByteArrayInputStream(rep.getConteudo());
        OutputStream out = response.getOutputStream();

        byte[] buffer = new byte[2048];//Buffer para leitura
        int lidos;

        while ((lidos = in.read(buffer))!= -1)
        {
            out.write(buffer,0,lidos);//Transfere imagem
            out.flush();
        }
        in.close();
        out.close();
    }


    // <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code.">
    /** 
     * Handles the HTTP <code>GET</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException {
        processRequest(request, response);
    } 

    /** 
     * Handles the HTTP <code>POST</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException {
        processRequest(request, response);
    }

    /** 
     * Returns a short description of the servlet.
     * @return a String containing servlet description
     */
    @Override
    public String getServletInfo() {
        return "Short description";
    }// </editor-fold>

}
